Sort Colors - Quicksort Partition - Leetcode 75 - Python